quick python program